package main

import (
	"client/proc"
	"fmt"
)

var (
	userId int
	userName string
	userPasswd string
)

func main() {
	var key byte

	for loop := true; loop; {
		fmt.Println("------------welcome to use char room software---------")
		fmt.Println("\t\t\t 1. 注册用户")
		fmt.Println("\t\t\t 2. 登录聊天室")
		fmt.Println("\t\t\t 3. 退出")
		fmt.Print("请输入你的选择： ")
		fmt.Scanf("%d\n", &key)

		switch key {
		case 1:
			fmt.Println("\t\t ------------注册新用户--------------")
			fmt.Print("请输入用户id：")
			fmt.Scanf("%d\n", &userId)

			fmt.Print("请输入用户名：")
			fmt.Scanf("%s\n", &userName)

			fmt.Print("请输入用户密码：")
			fmt.Scanf("%s\n", &userPasswd)

			up := &proc.UserProc{}

			err := up.Register(userId, userName, userPasswd)
			if err != nil {
				fmt.Println("main register ", err)
			} else {
				fmt.Println("\t\t注册成功，请登录....")
			}
		case 2:
			fmt.Println("\t\t ------------登录聊天室--------------")

			fmt.Print("请输入用户id：")
			fmt.Scanf("%d\n", &userId)

			fmt.Print("请输入用户密码：")
			fmt.Scanf("%s\n", &userPasswd)

			up := &proc.UserProc{}

			err := up.Login(userId, userPasswd)
			if err != nil {
				fmt.Println(err)
			} else {

			}
		case 3:
			fmt.Print("您是确定退出本系统（y/n）: ")
			var choice byte
			for {
				fmt.Scanf("%c\n", &choice)

				if choice == 'y' || choice == 'n' {
					if choice == 'y' {
						loop = false
					}
					break
				} else {
					fmt.Print("输入错误，请重新输入（y/n)： ")
				}
			}
		default:
			fmt.Println("输入错误，请重新输入")
		}
	}
	fmt.Println("您已退出本系统")
}
